[2주차 Stack & Hash] 나머지 한 점 - Lv1

문제 설명

직사각형을 만드는 데 필요한 4개의 점 중 3개의 좌표가 주어질 때, 나머지 한 점의 좌표를 구하려고 합니다. 점 3개의 좌표가 들어있는 배열 v가 매개변수로 주어질 때, 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 return 하도록 solution 함수를 완성해주세요. 단, 직사각형의 각 변은 x축, y축에 평행하며, 반드시 직사각형을 만들 수 있는 경우만 입력으로 주어집니다.

제한사항

• v는 세 점의 좌표가 들어있는 2차원 배열입니다.
• v의 각 원소는 점의 좌표를 나타내며, 좌표는 [x축 좌표, y축 좌표] 순으로 주어집니다.
• 좌표값은 1 이상 10억 이하의 자연수입니다.
• 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 [x축 좌표, y축 좌표] 순으로 담아 return 해주세요.

입출력 예

입출력 예 설명

입출력 예 #1 세 점이 [1, 4], [3, 4], [3, 10] 위치에 있을 때, [1, 10]에 점이 위치하면 직사각형이 됩니다.

입출력 예 #2 세 점이 [1, 1], [2, 2], [1, 2] 위치에 있을 때, [2, 1]에 점이 위치하면 직사각형이 됩니다.

나의 풀이

첫번째 풀이

소스

def solution(v):
    answer = []

    left_xpos = []
    left_ypos = []

    # Toggle 형식으로 홀수 개수인 점을 찾는다.
    for xpos, ypos in v:
        if xpos in left_xpos:
            left_xpos.remove(xpos)
        else:
            left_xpos.append(xpos)

        if ypos in left_ypos:
            left_ypos.remove(ypos)
        else:
            left_ypos.append(ypos)

    return left_xpos + left_ypos

설명

• hash나 stack을 이용하지않고 리스트로 풀었습니다.
• 직사각형이 되려면 각 위치좌표가 각각 2번씩 나와야 된다는 점을 이용하였습니다.

결과

통과

리뷰

사실 입력 값이 너무 작아서 큰 차이는 없습니다. :) 하지만 파이썬의 Counter를 이용해서 풀면 좀 더 깔끔하게 풀 수 있습니다. 한 번 도전해보세요!

두번째 풀이

소스

from collections import Counter 

def solution(v):
    answer = []

    x = [x for x, _ in v]
    y = [y for _, y in v]

    answer += [key for key, value in Counter(x).items() if value == 1]
    answer += [key for key, value in Counter(y).items() if value == 1]

    return answer 

설명

• 앞서 풀었던 아이디어를 기반으로 Counter로 풀어보았다.
• 훨씬 깔끔하고 pythonic하다.

결과

통과